# page/page_financial_tools.py
import time
from base.base import Base
from loc.financial_tools_page_locators import FinancialToolsPageLocators

class PageFinancialTools(Base):
    def calculate(self, amount, term):
        """执行计算操作"""
        self.click(FinancialToolsPageLocators.SUB_MENU_ITEM_1)
        time.sleep(1) # 临时保留，最好用显式等待替代
        self.click(FinancialToolsPageLocators.SUB_MENU_ITEM_2)
        time.sleep(1)
        self.input_text(FinancialToolsPageLocators.AMOUNT_INPUT, amount)
        self.input_text(FinancialToolsPageLocators.TERM_INPUT, term)
        self.click(FinancialToolsPageLocators.CALCULATE_BUTTON)
        time.sleep(1)
        self.click(FinancialToolsPageLocators.RESULT_POPUP_CONFIRM_BUTTON)
        return self

    def get_calculation_result(self):
        """获取计算结果用于断言"""
        # return self.get_text(FinancialToolsPageLocators.CALCULATION_RESULT_TEXT)
        # 实际项目中，这里应该获取结果并返回
        pass